package com.imooc.utils;

import com.alibaba.excel.context.AnalysisContext;
import com.alibaba.excel.event.AnalysisEventListener;

import java.util.*;

/**
 * @Description
 * @Author 顾志杰
 * @Date 2022/12/21 16:02
 */
public class ExcelListen extends AnalysisEventListener<Map> {
  String[] code = new String[]{"01","02","03","04","05","06","07","08","09",
    "10","11","12","13","14","15","16","17","18","19",
    "20","21","22","23","24","25","26","27","28","29",
    "30","31","32","33","34","35","36","37","38","39",
    "40","41","42","43","44","45","46","47","48","49",
    "40","41","42","43","44","45","46","47","48","49",
    "50","51","52","53","54","55","56","57","58","59",
    "60","61","62","63","64","65","66","67","68","69",
    "70","71","72","73","74","75","76","77","78","79",
    "80","81","82","83","84","85","86","87","88","89"};

  private List<Map<String,Object>> list = new ArrayList<>();

  @Override
  public void invoke(Map data, AnalysisContext context) {
    try {
      boolean ifContain = Arrays.toString(code).contains((CharSequence) data.get(1));
      if(ifContain){
        Map<String,Object> map = new HashMap<String,Object>(){{
          put(String.valueOf(data.get(1)),data.get(3));
        }};
        list.add(map);
      }
    } catch (Exception e) {

    }
  }

  @Override
  public void doAfterAllAnalysed(AnalysisContext context) {

  }

  public List<Map<String,Object>> getData(){
    return list;
  }
}
